package com.rock.code.system.dept.controller;

import com.alibaba.fastjson.JSONObject;
import com.github.pagehelper.PageInfo;
import com.rock.code.base.controller.BaseController;
import com.rock.code.base.domain.AjaxJson;
import com.rock.code.base.domain.Constants;
import com.rock.code.common.excel.ExportExcel;
import com.rock.code.common.exceptions.SystemException;
import com.rock.code.common.pager.Page;
import com.rock.code.common.tree.ItemNode;
import com.rock.code.system.dept.domain.TbSysDept;
import com.rock.code.system.dept.service.TbSysDeptService;
import com.rock.code.system.user.domain.TbSysLoginmsg;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* 部门管理
*
* @author kongqiang
* @date 2018/06/13
*/
@Controller
@RequestMapping("/tbSysDept")
public class TbSysDeptController extends BaseController {

    /**
    * The tbSysDeptServiceImpl
    */
    @Resource
    private TbSysDeptService tbSysDeptServiceImpl;

    /**
    * Dispatch.
    *
    * @return the string
    * @throws Exception the exception
    * @author kongqiang
    * @date 2018/06/13
    */
    @RequestMapping("")
    public String dispatch() throws Exception {
        return "system/dept/tbSysDeptList";
    }

    /**
    * queryPageList
    *
    * @return the page info
    * @throws Exception the exception
    * @author kongqiang
    * @date 2018/06/13
    */
    @ResponseBody
    @RequestMapping(value = "/queryPageList", method = RequestMethod.POST)
    public PageInfo<TbSysDept> queryPageList() throws Exception {
        Map<String, Object> param = this.getRequestMapSingle();
        Page page = new Page(request);
        PageInfo pageInfo = null;
        try {
            param.put("isDelete", Constants.DELETE_N);
            pageInfo = tbSysDeptServiceImpl.queryPageList(page, param);
        } catch (Exception e) {
            e.printStackTrace();
            throw new SystemException(e);
        }
        return pageInfo;
    }

    /**
    * querySingle
    *
    * @return TbSysDept
    * @throws Exception the exception
    * @author kongqiang
    * @date 2018/06/13
    */
    @ResponseBody
    @RequestMapping(value = "/querySingle", method = RequestMethod.POST)
    public TbSysDept querySingle() throws Exception {
        Map<String, Object> param = this.getRequestMapSingle();
        TbSysDept tbSysDept = null;
        try {
            tbSysDept = tbSysDeptServiceImpl.queryByParaMap(param);
        } catch (Exception e) {
            e.printStackTrace();
            throw new SystemException(e);
        }
        return tbSysDept;
    }

    /**
     * Query tree data.
     *
     * @param parentCode the input name
     * @return the json object
     * @throws Exception the exception
     * @author kongqiang
     * @date 2018/06/13
     */
    @ResponseBody
    @RequestMapping(value = "/queryTreeData", method = RequestMethod.GET)
    public AjaxJson queryTreeData(String parentCode) throws Exception {
        try {
            Map<String, Object> param = new HashMap<String, Object>();
            param.put("parentCode", parentCode);
            param.put("isDelete", Constants.DELETE_N);
            param.put("orderBy", "DEPT_CODE asc");
            List<ItemNode> rows = tbSysDeptServiceImpl.queryTreeData(param);
            ajaxJson.setResult(rows);
        } catch (Exception e) {
            e.printStackTrace();
            throw new SystemException(e);
        }
        return ajaxJson;
    }

    /**
    * Save.
    *
    * @param tbSysDept the TbSysDept
    * @return the ajax json
    * @throws Exception the exception
    * @author kongqiang
    * @date 2018/06/13
    */
    @ResponseBody
    @RequestMapping(value = "/save", method = RequestMethod.POST)
    public AjaxJson save(@RequestBody TbSysDept tbSysDept) throws Exception {
        try {
            TbSysLoginmsg sessionUser = getSessionUser();

            // 设置其他默认值
            tbSysDept.setSuperDeptId(tbSysDept.getParentCode());
            tbSysDept.setCompId(tbSysDept.getParentCode());

            if (isNotEmpty(tbSysDept.getDeptId())) {
                tbSysDept.setModifier(sessionUser.getLoginCode());
                tbSysDept.setModifieDate(new Date());
                tbSysDeptServiceImpl.update(tbSysDept);
            } else {
                // 根据父节点设置子节点的级别
                TbSysDept parentNode = null;
                Map<String, Object> map = new HashMap<>();
                map.put("deptId", tbSysDept.getParentCode());
                map.put("isDelete", Constants.DELETE_N);
                parentNode = tbSysDeptServiceImpl.queryByParaMap(map);
                tbSysDept.setTreeLevel(parentNode.getTreeLevel() + 1);
                // 设置子节点其他属性
                tbSysDept.setIsDelete(Constants.DELETE_N);
                tbSysDept.setCreater(sessionUser.getLoginCode());
                tbSysDept.setCreateDate(new Date());
                tbSysDept.setModifier(sessionUser.getLoginCode());
                tbSysDept.setModifieDate(new Date());
                tbSysDeptServiceImpl.insert(tbSysDept);
                // 更新父节点
                if (parentNode.getIsLeaf() == 1) {
                    parentNode.setIsLeaf(0);
                    tbSysDeptServiceImpl.update(parentNode);
                }
            }
            return ajaxJson;
        } catch (Exception e) {
            e.printStackTrace();
            throw new SystemException(e);
        }
    }

    /**
    * Delete.
    *
    * @param ids the ids
    * @return the ajax json
    * @throws Exception the exception
    * @author kongqiang
    * @date 2018/06/13
    */
    @ResponseBody
    @RequestMapping(value = "/delete", method = RequestMethod.POST)
    public AjaxJson delete(@RequestParam String ids) throws Exception {
        try {
            String[] deptIds = ids.split(",");
            TbSysLoginmsg sessionUser = getSessionUser();
            for (String deptId : deptIds) {
                // 根据id查找子节点
                Map<String, Object> param = new HashMap<>();
                param.put("deptId", deptId);
                param.put("isDelete", Constants.DELETE_N);
                TbSysDept node = tbSysDeptServiceImpl.queryByParaMap(param);
                // 删除子节点
                TbSysDept tbSysDept = new TbSysDept();
                tbSysDept.setDeptId(deptId);
                tbSysDept.setIsDelete(Constants.DELETE_Y);
                tbSysDept.setModifier(sessionUser.getLoginCode());
                tbSysDept.setModifieDate(new Date());
                tbSysDeptServiceImpl.update(tbSysDept);
                // 根据子节点的parentCode查询子节点数目
                Map<String, Object> rowParam = new HashMap<String, Object>();
                rowParam.put("parentCode", node.getParentCode());
                rowParam.put("isDelete", Constants.DELETE_N);
                List<ItemNode> rows = tbSysDeptServiceImpl.queryTreeData(rowParam);
                if (rows.size() == 0) {
                    // 更新父节点
                    Map<String, Object> parentParam = new HashMap<>();
                    parentParam.put("2", node.getParentCode());
                    parentParam.put("isDelete", Constants.DELETE_N);
                    TbSysDept parentNode = tbSysDeptServiceImpl.queryByParaMap(parentParam);
                    parentNode.setIsLeaf(new Integer(1));
                    tbSysDeptServiceImpl.update(parentNode);
                }
            }
            return ajaxJson;
        } catch (Exception e) {
            e.printStackTrace();
            throw new SystemException(e);
        }
    }

    /**
    * Exp excel.
    *
    * @return the ajax json
    * @throws Exception the exception
    * @author kongqiang
    * @date 2018/06/13
    */
    @ResponseBody
    @RequestMapping(value = "/expExcel", method = RequestMethod.POST)
    public AjaxJson expExcel() throws Exception {
        try {
            Map<String, Object> param = this.getRequestMapSingle();
            String[] titles = (String[]) param.get("titles[]");
            String[] fields = (String[]) param.get("fields[]");
            Page page = new Page((String) param.get("sidx"), (String) param.get("sord"));
            param.put("isDelete", Constants.DELETE_N);
            ExportExcel ee = new ExportExcel();
            List rows = tbSysDeptServiceImpl.queryListByOrder(page, param);
            String path = ee.exportExcel(request, response, 40000, titles, fields, rows, TbSysDept.class.getName());
            ajaxJson.setResult(path);
        } catch (Exception e) {
            e.printStackTrace();
            throw new SystemException(e);
        }
        return ajaxJson;
    }

    /**
    * Check input field.
    *
    * @param inputName the input name
    * @param inputValue the input value
    * @return the json object
    * @throws Exception the exception
    * @author kongqiang
    * @date 2018/06/13
    */
    @ResponseBody
    @RequestMapping(value = "/checkInputField", method = RequestMethod.POST)
    public JSONObject checkInputField(String inputName, String inputValue) throws Exception {
        try {
            Map<String, Object> param = new HashMap<String, Object>();
            param.put(inputName, inputValue);
            param.put("isDelete", Constants.DELETE_N);
            List rows = tbSysDeptServiceImpl.queryListByParaMap(param);
            if (rows != null && rows.size() > 0) {
                json.put("valid", false);
            } else {
                json.put("valid", true);
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new SystemException(e);
        }
        return json;
    }
}
